"""
Copyright (c) 2022 Huawei Technologies Co.,Ltd.

openGauss is licensed under Mulan PSL v2.
You can use this software according to the terms and conditions of the Mulan PSL v2.
You may obtain a copy of Mulan PSL v2 at:

          http://license.coscl.org.cn/MulanPSL2

THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
See the Mulan PSL v2 for more details.
"""
"""
Case Type   : sqlines工具
Case Name   : sqlines工具-s参数指定oracle，创建存储过程，查看转换情况
Create At   : 2022/01/12
@zou_jialiang0501115624
Description :
    1、sqlines安装
    2、创建测试脚本
    3、sqlines -s=oracle -t=opengauss -in={self.in_file} -out={self.out_file}
     -log={self.log_file}
    4、环境清理：卸载sqlines、删除对应文件
Expect      :
    1、sqlines安装成功
    2、脚本创建成功
    3、转换成功，自动添加or replace关键字、AS后与函数结尾，自动添加 $$
       尾部添加 language plpgsql、DBMS_OUTPUT.PUT_LINE ~ RAISE NOTICE
       传参操作符 => 会转换为 :=
    4、环境清理成功
History     :
    modified: 2023-04-19 by @daiguatutu 优化unzip解压命令，避免命令执行超时失败
"""

import os
import unittest

from yat.test import Node
from yat.test import macro
from testcase.utils.Logger import Logger


class SqlinesTest(unittest.TestCase):

    def setUp(self):
        self.log = Logger()
        self.log.info(f'----- {os.path.basename(__file__)} start-----')
        self.root = Node('PrimaryRoot')
        self.in_file = f'{os.path.basename(__file__)[:-3]}.sql'
        self.out_file = f'{os.path.basename(__file__)[:-3]}_out.sql'
        self.log_file = f'{os.path.basename(__file__)[:-3]}.log'
        self.file_name = 'openGauss-tools-sqlines-master.zip'

    def test_main(self):
        step_txt = '----step1:安装sqlines，expect: sqlines安装成功----'
        self.log.info(step_txt)
        self.log.info('-----下载软件包------')
        file_download = os.path.join(macro.FTP_PLUGINS_PATH,
                                     'sqlines', self.file_name)
        cmd = f'wget {file_download}'
        self.log.info(cmd)
        self.root.sh(cmd)
        cmd2 = f'ls {self.file_name}'
        self.log.info(cmd2)
        res = self.root.sh(cmd2).result()
        self.log.info(res)
        self.assertEqual(self.file_name, res, '文件下载失败：' + step_txt)

        self.log.info('-----解压软件包------')
        cmd3 = f'timeout 30s unzip -o {self.file_name}'
        self.log.info(cmd3)
        res = self.root.sh(cmd3).result()
        self.log.info(res)
        self.unzip_dir = self.file_name.replace('.zip', '')
        cmd4 = f'ls -d {self.unzip_dir}'
        self.log.info(cmd4)
        res = self.root.sh(cmd4).result()
        self.log.info(res)
        self.assertEqual(self.unzip_dir, res, '文件解压失败：' + step_txt)

        self.log.info('-----安装sqlines------')
        cmd = f'cd {self.unzip_dir};sh build.sh -i;pwd'
        self.log.info(cmd)
        res = self.root.sh(cmd).result()
        self.log.info(res)
        self.assertIn('Install Sqlines successfully', res,
                      'Sqlines安装失败：' + step_txt)
        tmp_path = res.strip().splitlines()[-1].strip()
        mid_path = os.path.join(tmp_path, 'bin')
        file_name = 'sqlines'
        check_file = os.path.join(mid_path, file_name)
        cmd = f"ls -l {check_file}"
        self.log.info(cmd)
        res = self.root.sh(cmd).result()
        self.log.info(res)
        self.assertEqual(res[3], 'x', '文件没有执行权限' + step_txt)
        self.log.info('-----配置sqlines环境变量-----')
        self.source_path = f"export PATH={mid_path}:$PATH"

        step_txt = '----step2:创建测试脚本，expect: 脚本创建成功----'
        self.log.info(step_txt)
        cmd1 = f"""cat >{self.in_file} <<EOF
        CREATE PROCEDURE PRO_Opengauss_Function_Sqlines_Case0029
         (p_deptno NUMBER) AS
          v_sal NUMBER(6, 2);
        BEGIN
          SELECT avg(SALARY) INTO v_sal FROM emp 
            WHERE DEPARTMENT_ID = p_deptno;
          DBMS_OUTPUT.PUT_LINE('dep'||p_deptno||' average sal is '||v_sal);
        END PRO_Opengauss_Function_Sqlines_Case0029;
        
        declare 
          p_deptno number;
        begin
          PRO_Opengauss_Function_Sqlines_Case0029(p_deptno => 100);
        end;
        """
        self.log.info(cmd1)
        self.root.sh(cmd1)
        cmd2 = f'ls {self.in_file}'
        self.log.info(cmd2)
        res = self.root.sh(cmd2).result()
        self.log.info(res)
        self.assertEqual(res, f'{self.in_file}', '执行失败' + step_txt)

        step_txt = '----step3:执行sqlines命令，expect: 转换成功，' \
                   '自动添加or replace关键字、' \
                   'AS后与函数结尾，自动添加 $$、尾部添加 language plpgsql' \
                   'DBMS_OUTPUT.PUT_LINE 转为 RAISE NOTICE' \
                   '传参操作符 => 会转换为 :=----'
        self.log.info(step_txt)
        cmd = f"{self.source_path};" \
            f"sqlines -s=oracle -t=opengauss -in={self.in_file} " \
            f"-out={self.out_file} -log={self.log_file}"
        self.log.info(cmd)
        res = self.root.sh(cmd).result()
        self.log.info(res)
        self.assertNotIn('error', res, '执行失败：')
        cmd2 = f'cat {self.out_file}'
        self.log.info(cmd2)
        res = self.root.sh(cmd2).result()
        self.log.info(res)
        self.assertTrue(len(res.strip().splitlines()) >= 15, '行数不到15行')

        check1 = res.strip().splitlines()[0].strip()
        check2 = res.strip().splitlines()[1].strip()
        check3 = res.strip().splitlines()[7].strip()
        check4 = res.strip().splitlines()[9].strip()
        check5 = res.strip().splitlines()[14].strip()

        self.assertIn('OR REPLACE', check1, '执行失败：' + step_txt)
        self.assertIn('AS $$', check2, '执行失败：' + step_txt)
        self.assertIn('RAISE NOTICE', check3, '执行失败：' + step_txt)
        self.assertIn('$$', check4, '执行失败：' + step_txt)
        self.assertIn('LANGUAGE plpgsql', check4, '执行失败：' + step_txt)
        self.assertIn('p_deptno :=', check5, '执行失败：' + step_txt)

    def tearDown(self):
        step_txt = '----step4:环境清理，expect: 环境清理成功----'
        self.log.info(step_txt)
        self.log.info('------卸载sqlines------')
        cmd = f'cd {self.unzip_dir};sh build.sh -m'
        self.log.info(cmd)
        res = self.root.sh(cmd).result()
        self.log.info(res)
        self.log.info('-----删除相关文件-----')
        cmd = f'rm -rf {self.file_name} {self.unzip_dir} sqlines.log ' \
                    f'{self.in_file} {self.out_file} {self.log_file};' \
                    f'if [ -d {self.file_name} ]; ' \
                    f'then echo "exists"; else echo "not exists"; fi'
        self.root.sh(cmd)
        self.assertIn('Uninstall Sqlines successfully', res,
                      'Sqlines卸载失败：' + step_txt)
        self.assertIn('not exists', cmd, '删除文件失败' + cmd)
        self.log.info(f'----- {os.path.basename(__file__)} end-----')
